/******************************************************************************/
/* @ All Rights Reserved Copyright (C) 2014 broada,Ltd. 		              */
/*																              */
/*	NAME	  = t__sqlserver_bigint_add.c						              */
/*	NOTE	  = 												              */
/*	DATE	  = 2014/11/25 by geyh								              */
/******************************************************************************/

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <assert.h>
#include <CUnit/Basic.h>
#include <CUnit/CUnit.h>

#include "apmtest_cunit.h"
#include "../test_db2_comm.h"
#include <db2/db2_core.h>
#include <db2/db2_comm.h>
#include <db2/db2_param.h>
#include <db2/db2_obj.h>
#include <db2/db2_error_msg.h>
#include <db2/db2_reply_msg.h>
#include <apmtest_comm.h>
#include <db2/config.h>

extern int __db2_parse_excsqlset(Conntrack *p_st_conn, Frhash_header *p_st_hash_report,
                                 L7session_db2_private *p_st_private, int i_whole_flg);

extern int __db2_parse_excsqlset_resp(Conntrack *p_st_conn, Frhash_header *p_st_hash_report,
                                      L7session_db2_private *p_st_private, int i_whole_flg, Db2_req_cmd *p_st_cmd_cp);

static int __suite_init(void)
{

    return 0;
}

static int __suite_cleanup(void)
{

    return 0;
}

static void __case1(void)
{
    int i_req_cp = 0;
    int i_ret = APM_L7_ANALYZE_SUCC;
    Frlist_node *p_st_cmd_node = NULL;
    Db2_req_cmd *p_st_cmd_cp = NULL;

    Conntrack *p_st_conn = NULL;
    CU_ASSERT_EQUAL(
            test_db2_create_ct(&p_st_conn, TEST_DB2_SRC_IP, TEST_DB2_DST_IP, TEST_DB2_SRC_PORT, TEST_DB2_DST_PORT),
            FR_SUCC);
    L7_session *p_st_session = frct_get_session(p_st_conn);
    L7session_db2_private *p_st_private = (L7session_db2_private *) frct_get_session_private(p_st_session);

    Frhash_header st_hash_report;
    CU_ASSERT_EQUAL(frhash2_create(&st_hash_report, 3), FR_SUCC);

    p_st_private->p_st_req = db2_create_request();
    p_st_private->p_st_fragment->p_st_data = bytebuff_creat();
    p_st_private->p_st_fragment->st_start_timestamp.tv_sec = 1;
    p_st_private->p_st_fragment->st_start_timestamp.tv_usec = 1;
    p_st_private->p_st_fragment->st_timestamp.tv_sec = 1;
    p_st_private->p_st_fragment->st_timestamp.tv_usec = 1;
    bytebuff_add(p_st_private->p_st_fragment->p_st_data,
                 (char *) "\x00\x53\xd0\x51\x00\x01\x00\x4d\x20\x14\x00\x44\x21\x13\xe3\xd6\xd6\xd3\xe2\xc4\xc2\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40\xd5\xe4\xd3\xd3\xc9\xc4\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40\xe2\xe8\xe2\xe2\xc8\xf2\xf0\xf0\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40\x53\x59\x53\x4c\x56\x4c\x30\x31\x00\x41\x00\x05\x21\x0e\x01\x00\x5f\xd0\x43\x00\x01\x00\x59\x24\x14\x00\x00\x00\x00\x4f\x53\x45\x54\x20\x43\x4c\x49\x45\x4e\x54\x20\x41\x43\x43\x54\x4e\x47\x27\x4a\x43\x43\x30\x32\x31\x30\x30\x57\x69\x6e\x2d\x50\x43\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x61\x64\x6d\x69\x6e\x69\x73\x74\x27\x2c\x58\x27\x30\x30\x27\xff",
                 178);
    bytebuff_setPosition(p_st_private->p_st_fragment->p_st_data, p_st_private->p_st_fragment->u64_parsed_index);
    db2_get_ddm_cp(p_st_private->p_st_fragment, DB2_FALSE, &i_req_cp, &p_st_private->p_st_req->ui_ddm_len);
    CU_ASSERT_EQUAL(i_req_cp, DRDA_CP_EXCSQLSET);

    i_ret = __db2_parse_excsqlset(p_st_conn, &st_hash_report, p_st_private, DB2_FALSE);

    CU_ASSERT_EQUAL(i_ret, APM_L7_ANALYZE_SUCC);

    db2_init_fragment(p_st_private->p_st_fragment);
    p_st_private->p_st_fragment->st_start_timestamp.tv_sec = 2;
    p_st_private->p_st_fragment->st_start_timestamp.tv_usec = 2;
    p_st_private->p_st_fragment->st_timestamp.tv_sec = 2;
    p_st_private->p_st_fragment->st_timestamp.tv_usec = 2;
    p_st_private->p_st_req->i_need_req_resp = DB2_RESPONSE;
    /* response */
    p_st_private->p_st_resp = db2_create_response();
    p_st_private->p_st_fragment->p_st_data = bytebuff_creat();
    bytebuff_add(p_st_private->p_st_fragment->p_st_data, (char *) "\x00\x0b\xd0\x43\x00\x01\x00\x05\x24\x08\xff", 11);
    bytebuff_setPosition(p_st_private->p_st_fragment->p_st_data, p_st_private->p_st_fragment->u64_parsed_index);
    /* 取得当前解析到对应request的command */
    p_st_cmd_node = frlist_node_idx(&p_st_private->p_st_req->st_command_lst, 0);
    p_st_cmd_cp = (Db2_req_cmd *) FRLIST_GET_NODE_DATA(p_st_cmd_node);

    p_st_conn->flg_direct_known = FR_DIRECT_KNOWN;

    i_ret = __db2_parse_excsqlset_resp(p_st_conn, &st_hash_report, p_st_private, DB2_FALSE, p_st_cmd_cp);

    CU_ASSERT_EQUAL(i_ret, APM_L7_ANALYZE_SUCC);

    frhash2_free(&st_hash_report, list_report_node_free);
    test_db2_destroy_ct(&p_st_conn);

    return;
}

static void __case2(void)
{
    int i_req_cp = 0;
    int i_ret = APM_L7_ANALYZE_SUCC;
    Frlist_node *p_st_cmd_node = NULL;
    Db2_req_cmd *p_st_cmd_cp = NULL;

    Conntrack *p_st_conn = NULL;
    CU_ASSERT_EQUAL(
            test_db2_create_ct(&p_st_conn, TEST_DB2_SRC_IP, TEST_DB2_DST_IP, TEST_DB2_SRC_PORT, TEST_DB2_DST_PORT),
            FR_SUCC);
    L7_session *p_st_session = frct_get_session(p_st_conn);
    L7session_db2_private *p_st_private = (L7session_db2_private *) frct_get_session_private(p_st_session);

    Frhash_header st_hash_report;
    CU_ASSERT_EQUAL(frhash2_create(&st_hash_report, 3), FR_SUCC);

    p_st_private->p_st_req = db2_create_request();
    p_st_private->p_st_fragment->p_st_data = bytebuff_creat();
    p_st_private->p_st_fragment->st_start_timestamp.tv_sec = 1;
    p_st_private->p_st_fragment->st_start_timestamp.tv_usec = 1;
    p_st_private->p_st_fragment->st_timestamp.tv_sec = 1;
    p_st_private->p_st_fragment->st_timestamp.tv_usec = 1;
    bytebuff_add(p_st_private->p_st_fragment->p_st_data,
                 (char *) "\x00\x53\xd0\x51\x00\x01\x00\x4d\x20\x14\x00\x44\x21\x13\xe3\xd6\xd6\xd3\xe2\xc4\xc2\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40\xd5\xe4\xd3\xd3\xc9\xc4\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40\xe2\xe8\xe2\xe2\xc8\xf2\xf0\xf0\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40\x53\x59\x53\x4c\x56\x4c\x30\x31\x00\x41\x00\x05\x21\x0e\x01\x00\x5f\xd0\x43\x00\x01\x00\x59\x24\x14\x00\x00\x00\x00\x4f\x53\x45\x54\x20\x43\x4c\x49\x45\x4e\x54\x20\x41\x43\x43\x54\x4e\x47\x27\x4a\x43\x43\x30\x32\x31\x30\x30\x57\x69\x6e\x2d\x50\x43\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x61\x64\x6d\x69\x6e\x69\x73\x74\x27\x2c\x58\x27\x30\x30\x27\xff",
                 178);
    bytebuff_setPosition(p_st_private->p_st_fragment->p_st_data, p_st_private->p_st_fragment->u64_parsed_index);
    db2_get_ddm_cp(p_st_private->p_st_fragment, DB2_FALSE, &i_req_cp, &p_st_private->p_st_req->ui_ddm_len);
    CU_ASSERT_EQUAL(i_req_cp, DRDA_CP_EXCSQLSET);

    i_ret = __db2_parse_excsqlset(p_st_conn, &st_hash_report, p_st_private, DB2_FALSE);

    CU_ASSERT_EQUAL(i_ret, APM_L7_ANALYZE_SUCC);

    db2_init_fragment(p_st_private->p_st_fragment);
    p_st_private->p_st_fragment->st_start_timestamp.tv_sec = 2;
    p_st_private->p_st_fragment->st_start_timestamp.tv_usec = 2;
    p_st_private->p_st_fragment->st_timestamp.tv_sec = 2;
    p_st_private->p_st_fragment->st_timestamp.tv_usec = 2;
    p_st_private->p_st_req->i_need_req_resp = DB2_RESPONSE;
    /* response */
    p_st_private->p_st_resp = db2_create_response();
    p_st_private->p_st_fragment->p_st_data = bytebuff_creat();
    bytebuff_add(p_st_private->p_st_fragment->p_st_data, (char *) "\x00\x0b\xd0\x43\x00\x01\x00\x05\x24\x08", 10);
    bytebuff_setPosition(p_st_private->p_st_fragment->p_st_data, p_st_private->p_st_fragment->u64_parsed_index);
    /* 取得当前解析到对应request的command */
    p_st_cmd_node = frlist_node_idx(&p_st_private->p_st_req->st_command_lst, 0);
    p_st_cmd_cp = (Db2_req_cmd *) FRLIST_GET_NODE_DATA(p_st_cmd_node);

    p_st_conn->flg_direct_known = FR_DIRECT_KNOWN;

    i_ret = __db2_parse_excsqlset_resp(p_st_conn, &st_hash_report, p_st_private, DB2_FALSE, p_st_cmd_cp);

    CU_ASSERT_EQUAL(i_ret, APM_L7_ANALYZE_TODO);

    frhash2_free(&st_hash_report, list_report_node_free);
    test_db2_destroy_ct(&p_st_conn);

    return;
}

static void __case3(void)
{
    int i_req_cp = 0;
    int i_ret = APM_L7_ANALYZE_SUCC;
    Frlist_node *p_st_cmd_node = NULL;
    Db2_req_cmd *p_st_cmd_cp = NULL;

    Conntrack *p_st_conn = NULL;
    CU_ASSERT_EQUAL(
            test_db2_create_ct(&p_st_conn, TEST_DB2_SRC_IP, TEST_DB2_DST_IP, TEST_DB2_SRC_PORT, TEST_DB2_DST_PORT),
            FR_SUCC);
    L7_session *p_st_session = frct_get_session(p_st_conn);
    L7session_db2_private *p_st_private = (L7session_db2_private *) frct_get_session_private(p_st_session);

    Frhash_header st_hash_report;
    CU_ASSERT_EQUAL(frhash2_create(&st_hash_report, 3), FR_SUCC);

    p_st_private->p_st_req = db2_create_request();
    p_st_private->p_st_fragment->p_st_data = bytebuff_creat();
    p_st_private->p_st_fragment->st_start_timestamp.tv_sec = 1;
    p_st_private->p_st_fragment->st_start_timestamp.tv_usec = 1;
    p_st_private->p_st_fragment->st_timestamp.tv_sec = 1;
    p_st_private->p_st_fragment->st_timestamp.tv_usec = 1;
    bytebuff_add(p_st_private->p_st_fragment->p_st_data,
                 (char *) "\x00\x53\xd0\x51\x00\x01\x00\x4d\x20\x14\x00\x44\x21\x13\xe3\xd6\xd6\xd3\xe2\xc4\xc2\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40\xd5\xe4\xd3\xd3\xc9\xc4\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40\xe2\xe8\xe2\xe2\xc8\xf2\xf0\xf0\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40\x53\x59\x53\x4c\x56\x4c\x30\x31\x00\x41\x00\x05\x21\x0e\x01\x00\x5f\xd0\x43\x00\x01\x00\x59\x24\x14\x00\x00\x00\x00\x4f\x53\x45\x54\x20\x43\x4c\x49\x45\x4e\x54\x20\x41\x43\x43\x54\x4e\x47\x27\x4a\x43\x43\x30\x32\x31\x30\x30\x57\x69\x6e\x2d\x50\x43\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x61\x64\x6d\x69\x6e\x69\x73\x74\x27\x2c\x58\x27\x30\x30\x27\xff",
                 178);
    bytebuff_setPosition(p_st_private->p_st_fragment->p_st_data, p_st_private->p_st_fragment->u64_parsed_index);
    db2_get_ddm_cp(p_st_private->p_st_fragment, DB2_FALSE, &i_req_cp, &p_st_private->p_st_req->ui_ddm_len);
    CU_ASSERT_EQUAL(i_req_cp, DRDA_CP_EXCSQLSET);

    i_ret = __db2_parse_excsqlset(p_st_conn, &st_hash_report, p_st_private, DB2_FALSE);

    CU_ASSERT_EQUAL(i_ret, APM_L7_ANALYZE_SUCC);

    db2_init_fragment(p_st_private->p_st_fragment);
    p_st_private->p_st_fragment->st_start_timestamp.tv_sec = 2;
    p_st_private->p_st_fragment->st_start_timestamp.tv_usec = 2;
    p_st_private->p_st_fragment->st_timestamp.tv_sec = 2;
    p_st_private->p_st_fragment->st_timestamp.tv_usec = 2;
    p_st_private->p_st_req->i_need_req_resp = DB2_RESPONSE;
    /* response */
    p_st_private->p_st_resp = db2_create_response();
    p_st_private->p_st_fragment->p_st_data = bytebuff_creat();
    bytebuff_add(p_st_private->p_st_fragment->p_st_data, (char *) "\x00\x0b\xd0\x43\x00\x01\x00\x05\x24\x08", 10);
    bytebuff_setPosition(p_st_private->p_st_fragment->p_st_data, p_st_private->p_st_fragment->u64_parsed_index);
    /* 取得当前解析到对应request的command */
    p_st_cmd_node = frlist_node_idx(&p_st_private->p_st_req->st_command_lst, 0);
    p_st_cmd_cp = (Db2_req_cmd *) FRLIST_GET_NODE_DATA(p_st_cmd_node);

    p_st_conn->flg_direct_known = FR_DIRECT_KNOWN;

    i_ret = __db2_parse_excsqlset_resp(p_st_conn, &st_hash_report, p_st_private, DB2_FALSE, p_st_cmd_cp);

    CU_ASSERT_EQUAL(i_ret, APM_L7_ANALYZE_TODO);

    p_st_private->p_st_fragment->st_timestamp.tv_sec = 3;
    p_st_private->p_st_fragment->st_timestamp.tv_usec = 3;
    bytebuff_add(p_st_private->p_st_fragment->p_st_data, (char *) "\xff", 1);
    bytebuff_setPosition(p_st_private->p_st_fragment->p_st_data, p_st_private->p_st_fragment->u64_parsed_index);
    /* 取得当前解析到对应request的command */
    p_st_cmd_node = frlist_node_idx(&p_st_private->p_st_req->st_command_lst, 0);
    p_st_cmd_cp = (Db2_req_cmd *) FRLIST_GET_NODE_DATA(p_st_cmd_node);

    i_ret = __db2_parse_excsqlset_resp(p_st_conn, &st_hash_report, p_st_private, DB2_FALSE, p_st_cmd_cp);

    CU_ASSERT_EQUAL(i_ret, APM_L7_ANALYZE_SUCC);

    frhash2_free(&st_hash_report, list_report_node_free);
    test_db2_destroy_ct(&p_st_conn);

    return;
}

static void __case4(void)
{
    int i_req_cp = 0;
    int i_ret = APM_L7_ANALYZE_SUCC;
    Frlist_node *p_st_cmd_node = NULL;
    Db2_req_cmd *p_st_cmd_cp = NULL;

    Conntrack *p_st_conn = NULL;
    CU_ASSERT_EQUAL(
            test_db2_create_ct(&p_st_conn, TEST_DB2_SRC_IP, TEST_DB2_DST_IP, TEST_DB2_SRC_PORT, TEST_DB2_DST_PORT),
            FR_SUCC);
    L7_session *p_st_session = frct_get_session(p_st_conn);
    L7session_db2_private *p_st_private = (L7session_db2_private *) frct_get_session_private(p_st_session);

    Frhash_header st_hash_report;
    CU_ASSERT_EQUAL(frhash2_create(&st_hash_report, 3), FR_SUCC);

    p_st_private->p_st_req = db2_create_request();
    p_st_private->p_st_fragment->p_st_data = bytebuff_creat();
    p_st_private->p_st_fragment->st_start_timestamp.tv_sec = 1;
    p_st_private->p_st_fragment->st_start_timestamp.tv_usec = 1;
    p_st_private->p_st_fragment->st_timestamp.tv_sec = 1;
    p_st_private->p_st_fragment->st_timestamp.tv_usec = 1;
    bytebuff_add(p_st_private->p_st_fragment->p_st_data,
                 (char *) "\x00\x53\xd0\x51\x00\x01\x00\x4d\x20\x14\x00\x44\x21\x13\xe3\xd6\xd6\xd3\xe2\xc4\xc2\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40\xd5\xe4\xd3\xd3\xc9\xc4\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40\xe2\xe8\xe2\xe2\xc8\xf2\xf0\xf0\x40\x40\x40\x40\x40\x40\x40\x40\x40\x40\x53\x59\x53\x4c\x56\x4c\x30\x31\x00\x41\x00\x05\x21\x0e\x01\x00\x5f\xd0\x43\x00\x01\x00\x59\x24\x14\x00\x00\x00\x00\x4f\x53\x45\x54\x20\x43\x4c\x49\x45\x4e\x54\x20\x41\x43\x43\x54\x4e\x47\x27\x4a\x43\x43\x30\x32\x31\x30\x30\x57\x69\x6e\x2d\x50\x43\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x20\x61\x64\x6d\x69\x6e\x69\x73\x74\x27\x2c\x58\x27\x30\x30\x27\xff",
                 178);
    bytebuff_setPosition(p_st_private->p_st_fragment->p_st_data, p_st_private->p_st_fragment->u64_parsed_index);
    db2_get_ddm_cp(p_st_private->p_st_fragment, DB2_FALSE, &i_req_cp, &p_st_private->p_st_req->ui_ddm_len);
    CU_ASSERT_EQUAL(i_req_cp, DRDA_CP_EXCSQLSET);

    i_ret = __db2_parse_excsqlset(p_st_conn, &st_hash_report, p_st_private, DB2_FALSE);

    CU_ASSERT_EQUAL(i_ret, APM_L7_ANALYZE_SUCC);

    db2_init_fragment(p_st_private->p_st_fragment);
    p_st_private->p_st_fragment->st_start_timestamp.tv_sec = 2;
    p_st_private->p_st_fragment->st_start_timestamp.tv_usec = 2;
    p_st_private->p_st_fragment->st_timestamp.tv_sec = 2;
    p_st_private->p_st_fragment->st_timestamp.tv_usec = 2;
    p_st_private->p_st_req->i_need_req_resp = DB2_RESPONSE;
    /* response */
    p_st_private->p_st_resp = db2_create_response();
    p_st_private->p_st_fragment->p_st_data = bytebuff_creat();
    bytebuff_add(p_st_private->p_st_fragment->p_st_data, (char *) "\x00\x0b\xd0\x43\x00\x01\x00\x06\x24\x08", 10);
    bytebuff_setPosition(p_st_private->p_st_fragment->p_st_data, p_st_private->p_st_fragment->u64_parsed_index);
    /* 取得当前解析到对应request的command */
    p_st_cmd_node = frlist_node_idx(&p_st_private->p_st_req->st_command_lst, 0);
    p_st_cmd_cp = (Db2_req_cmd *) FRLIST_GET_NODE_DATA(p_st_cmd_node);

    p_st_conn->flg_direct_known = FR_DIRECT_KNOWN;

    i_ret = __db2_parse_excsqlset_resp(p_st_conn, &st_hash_report, p_st_private, DB2_TRUE, p_st_cmd_cp);

    CU_ASSERT_EQUAL(i_ret, APM_L7_ANALYZE_FAIL);

    frhash2_free(&st_hash_report, list_report_node_free);
    test_db2_destroy_ct(&p_st_conn);

    return;
}

int add_suite__db2_parse_excsqlset_resp(void)
{
    CU_pSuite pSuite = NULL;

    if ((pSuite = CU_add_suite(MODULE_NAME" __db2_parse_excsqlset_resp()", __suite_init, __suite_cleanup)) == NULL)
        return FR_FAIL;

    if (CU_add_test(pSuite, "case1", __case1) == NULL ||
        CU_add_test(pSuite, "case2", __case2) == NULL ||
        CU_add_test(pSuite, "case3", __case3) == NULL ||
        CU_add_test(pSuite, "case4", __case4) == NULL)
        return FR_FAIL;

    return FR_SUCC;
}
